from datasets import load_dataset


dataset = load_dataset("imdb")
train_data = dataset["train"]
test_data = dataset["test"]
print(len(train_data))
print(len(test_data))
